"""
调用维基百科信息：
1.创建维基百科api包装器对象 WikipediaAPIWrapper
2.建立查询对象 WikipediaQueryRun
3.调用查询对象的run方法查询信息
https://www.wikipedia.org/ 主站
https://zh.wikipedia.org/ 中文
"""

from langchain_community.agent_toolkits.load_tools import get_all_tool_names
from langchain_community.tools.wikipedia.tool import WikipediaQueryRun,WikipediaAPIWrapper
from langchain_community.tools.wikidata.tool import WikidataQueryRun,WikidataAPIWrapper

print(get_all_tool_names())
exit()


api = WikipediaAPIWrapper(top_k_results=5,lang="zh")

run = WikipediaQueryRun(api_wrapper=api)

r = run.invoke("库里")
print(r)

print("-----------------")

api = WikidataAPIWrapper(top_k_results=5,lang="en")

run = WikidataQueryRun(api_wrapper=api)

r = run.invoke("库里")
print(r)